# -*- coding:utf-8 -*-

from django_filters.filterset import FilterSet
from django_filters import filters

from Apps.DataGeneration.models import DataGenerationCat, DataGeneration


class DataGenerationCatFilter(FilterSet):
    name = filters.CharFilter(field_name='name', lookup_expr='contains')

    class Meta:
        model = DataGenerationCat
        fields = ["name"]


class DataGenerationFilter(FilterSet):
    name = filters.CharFilter(field_name='name', lookup_expr='contains')

    class Meta:
        model = DataGeneration
        fields = ["name"]
